# -*- coding: utf-8 -*-
"""
@Time    : 2024/7/29 13:51 
@Author  : ZhangShenao 
@File    : 11.使用DoctranTextTranslator文档翻译器.py 
@Desc    : 使用DoctranTextTranslator文档翻译器,将文档翻译成特定语言
"""
import os

import dotenv
from langchain_community.document_loaders import UnstructuredFileLoader
from langchain_community.document_transformers import DoctranTextTranslator

# 加载环境变量
dotenv.load_dotenv()

# 加载文档
loader = UnstructuredFileLoader('./docs/工作总结.txt')
docs = loader.load()

# 创建DoctranTextTranslator文档翻译器
translator = DoctranTextTranslator(
    openai_api_key=os.getenv('OPENAI_API_KEY'),  # 指定OpenAI Key
    openai_api_model='gpt-3.5-turbo',  # 指定OpenAI模型
    language='english',  # 指定目标语言
)

# 翻译文档
translated_docs = translator.transform_documents(docs)
print(translated_docs[0].page_content)
